from Test3_vggnet.model import vgg

net= model_name = "vgg16"
net = vgg(model_name=model_name, num_classes=12, init_weights=True) # 注意：模型内部传参数和不传参数，输出的结果是不一样的

# net = MobileNetV2(num_classes=12)
# 计算网络参数
total = sum([param.nelement() for param in net.parameters()])
# 精确地计算：1MB=1024KB=1048576字节
print('Number of parameter: % .4fM' % (total / 1e6))



